package com.zhao.constant;

import lombok.extern.slf4j.Slf4j;

import java.lang.reflect.Field;
import java.lang.reflect.Modifier;
import java.math.BigDecimal;
import java.sql.Blob;
import java.sql.Clob;
import java.sql.Types;
import java.util.Date;
import java.util.Map;
import java.util.TreeMap;

/**
 * @Author:zhao
 * @Date: 2018/8/26 11:25
 * @Description:
 */
@SuppressWarnings("all")
@Slf4j
public class JDBCTypeConstant {

    private static Map<String, Integer> jdbcTypeValues;
    private static Map<Integer, Class<?>> jdbcJavaTypes;

    static {
        jdbcTypeValues= new TreeMap<String,Integer>();
        jdbcJavaTypes = new TreeMap<Integer, Class<?>>();

        Field[] fields = java.sql.Types.class.getFields();
        for (int i=0, len=fields.length; i<len; ++i) {
            if (Modifier.isStatic(fields[i].getModifiers())) {
                try {
                    String name = fields[i].getName();
                    Integer value = (Integer) fields[i].get(java.sql.Types.class);
                    jdbcTypeValues.put(name,value);
                } catch (Exception e){

                }
            }
        }

        // 初始化jdbcJavaTypes：
        jdbcJavaTypes.put(new Integer(Types.LONGNVARCHAR), String.class);  // -16 字符串
        jdbcJavaTypes.put(new Integer(Types.NCHAR), String.class);    // -15 字符串
        jdbcJavaTypes.put(new Integer(Types.NVARCHAR), String.class);   // -9 字符串
        jdbcJavaTypes.put(new Integer(Types.ROWID), String.class);    // -8 字符串
        jdbcJavaTypes.put(new Integer(Types.BIT), Boolean.class);    // -7 布尔
        jdbcJavaTypes.put(new Integer(Types.TINYINT), Byte.class);    // -6 数字
        jdbcJavaTypes.put(new Integer(Types.BIGINT), Long.class);    // -5 数字
        jdbcJavaTypes.put(new Integer(Types.LONGVARBINARY), Blob.class);  // -4 二进制
        jdbcJavaTypes.put(new Integer(Types.VARBINARY), Blob.class);   // -3 二进制
        jdbcJavaTypes.put(new Integer(Types.BINARY), Blob.class);    // -2 二进制
        jdbcJavaTypes.put(new Integer(Types.LONGVARCHAR), String.class);  // -1 字符串
        //  jdbcJavaTypes.put(new Integer(Types.NULL), String.class);    // 0 /
        jdbcJavaTypes.put(new Integer(Types.CHAR), String.class);    // 1 字符串
        jdbcJavaTypes.put(new Integer(Types.NUMERIC), BigDecimal.class);  // 2 数字
        jdbcJavaTypes.put(new Integer(Types.DECIMAL), BigDecimal.class);  // 3 数字
        jdbcJavaTypes.put(new Integer(Types.INTEGER), Integer.class);   // 4 数字
        jdbcJavaTypes.put(new Integer(Types.SMALLINT), Short.class);   // 5 数字
        jdbcJavaTypes.put(new Integer(Types.FLOAT), BigDecimal.class);   // 6 数字
        jdbcJavaTypes.put(new Integer(Types.REAL), BigDecimal.class);   // 7 数字
        jdbcJavaTypes.put(new Integer(Types.DOUBLE), BigDecimal.class);  // 8 数字
        jdbcJavaTypes.put(new Integer(Types.VARCHAR), String.class);   // 12 字符串
        jdbcJavaTypes.put(new Integer(Types.BOOLEAN), Boolean.class);   // 16 布尔
        //  jdbcJavaTypes.put(new Integer(Types.DATALINK), String.class);   // 70 /
        jdbcJavaTypes.put(new Integer(Types.DATE), Date.class);    // 91 日期
        jdbcJavaTypes.put(new Integer(Types.TIME), Date.class);    // 92 日期
        jdbcJavaTypes.put(new Integer(Types.TIMESTAMP), Date.class);   // 93 日期
        jdbcJavaTypes.put(new Integer(Types.OTHER), Object.class);    // 1111 其他类型？
        //  jdbcJavaTypes.put(new Integer(Types.JAVA_OBJECT), Object.class);  // 2000
        //  jdbcJavaTypes.put(new Integer(Types.DISTINCT), String.class);   // 2001
        //  jdbcJavaTypes.put(new Integer(Types.STRUCT), String.class);   // 2002
        //  jdbcJavaTypes.put(new Integer(Types.ARRAY), String.class);    // 2003
        jdbcJavaTypes.put(new Integer(Types.BLOB), Blob.class);    // 2004 二进制
        jdbcJavaTypes.put(new Integer(Types.CLOB), Clob.class);    // 2005 大文本
        //  jdbcJavaTypes.put(new Integer(Types.REF), String.class);    // 2006
        //  jdbcJavaTypes.put(new Integer(Types.SQLXML), String.class);   // 2009
        jdbcJavaTypes.put(new Integer(Types.NCLOB), Clob.class);    // 2011 大文本
    }

    /**
     * 通过DB列名获取Java类型名
     * @param jdbcType
     * @return
     */
    public static String getJavaType(String jdbcType) {
        try{
            Integer type=jdbcTypeValues.get(jdbcType);
            return jdbcJavaTypes.get(type)!=null?jdbcJavaTypes.get(type).getSimpleName():null;
        }catch (Exception e){
            log.error("[{}]属性自动转换失败,启用手动配置",jdbcType);
        }
        return getJavaTypeConfigType(jdbcType);
    }

    /**
     * 通过DB列名获取Java类型全路径
     * @param jdbcType
     * @return
     */
    public static String getJavaTypeFull(String jdbcType){
        try{
            Integer type=jdbcTypeValues.get(jdbcType);
            return jdbcJavaTypes.get(type)!=null?jdbcJavaTypes.get(type).getName():null;
        }catch (Exception e){
            //log.warn("[{}]导包自动转换失败,启用手动配置",jdbcType);
        }
        return getJavaTypeFullConfigType(jdbcType);
    }

    /**
     * 不在Type类中的数据库类型采取手动配置来映射Java类型名
     * @param jdbcType
     * @return
     */
    private static String getJavaTypeConfigType(String jdbcType){
        switch (jdbcType){
            case "DATETIME":
                return "LocalDateTime";
            case "INT":
                return "Integer";
            default:
                break;
        }
        return null;
    }

    /**
     * 不在Type类中的数据库类型采取手动配置来映射Java类型名
     * @param jdbcType
     * @return
     */
    private static String getJavaTypeFullConfigType(String jdbcType){
        switch (jdbcType){
            case "DATETIME":
                return "java.time.LocalDateTime";
            case "INT":
                return "java.lang.Integer";
            default:
                break;
        }
        return null;
    }

}
